package com.trace.app.util;

import cn.hutool.http.HttpUtil;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import lombok.extern.slf4j.Slf4j;
import org.apache.http.HttpEntity;
import org.apache.http.HttpResponse;
import org.apache.http.HttpStatus;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClients;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;

import java.io.*;
import java.nio.charset.Charset;
import java.util.HashMap;
import java.util.Map;

/**
 * @author : lyj
 * @Timer : 2024/3/15
 * @Description :
 */
@Slf4j
@Component
public class WxQRCodeUtil {
    @Value("${wx.app-id}")
    private String appId;

    @Value("${wx.app-key}")
    private String appKey;

    /**
     * 获取微信accessToken
     */
    public static String getWxAccessToken(String appId, String appKey) throws CustomizeException {
        String tokenUrl = "https://api.weixin.qq.com/cgi-bin/token?grant_type=client_credential&appid=" + appId + "&secret=" + appKey;

        try {
            String jsonStr = HttpUtil.get(tokenUrl);
            log.info("access_token:{}",jsonStr);
            JSONObject jsonObject = JSON.parseObject(jsonStr);
            if (jsonObject.containsKey("access_token")) {
                String accessToken = jsonObject.getString("access_token");
                return accessToken;
            } else {
                log.error("未获取到token");
                throw new CustomizeException("未获取到token");
            }

        } catch (Exception e) {
            log.error("未获取到token");
            throw new CustomizeException(e.getMessage());
        }
    }

    public static ByteArrayInputStream sendPost(String URL, String json) {
        InputStream inputStream = null;
        ByteArrayInputStream byteArrayInputStream = null;
        // 创建默认的httpClient实例.
        CloseableHttpClient httpclient = HttpClients.createDefault();
        // 创建httppost
        HttpPost httppost = new HttpPost(URL);
        httppost.addHeader("Content-type", "application/json; charset=utf-8");
        httppost.setHeader("Accept", "application/json");
        try {
            StringEntity s = new StringEntity(json, Charset.forName("UTF-8"));
            s.setContentEncoding("UTF-8");
            httppost.setEntity(s);
            HttpResponse response = httpclient.execute(httppost);
            if (response.getStatusLine().getStatusCode() == HttpStatus.SC_OK) {
                // 获取相应实体
                HttpEntity entity = response.getEntity();
                inputStream = entity.getContent();
                ByteArrayOutputStream outStream = new ByteArrayOutputStream();
                // 创建一个Buffer字符串
                byte[] buffer = new byte[1024];
                // 每次读取的字符串长度，如果为-1，代表全部读取完毕
                int len = 0;
                // 使用一个输入流从buffer里把数据读取出来
                while ((len = inputStream.read(buffer)) != -1) {
                    // 用输出流往buffer里写入数据，中间参数代表从哪个位置开始读，len代表读取的长度
                    outStream.write(buffer, 0, len);
                }
                // 关闭输入流
                inputStream.close();
                // 把outStream里的数据写入内存
                byteArrayInputStream = new ByteArrayInputStream(outStream.toByteArray());
            }
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            // 关闭连接,释放资源
            try {
                httpclient.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        return byteArrayInputStream;
    }

    /**
     * 获取小程序菊花码
     * parm：machineNo    二维码想携带的参数
     **/
    public static String getWXCode(String machineNo) throws CustomizeException {
        String appId = "wx98c8c10bfff556dc";   //自己小程序的appid
        String secret = "830d18c5df00630eba559ec7fe6d5d5f";   //自己小程序的密钥
        String aliYunUrl = "";
        try {
            //这里调用的是上面的获取access_token方法
            String access_token = getWxAccessToken(appId, secret);

            String url = "https://api.weixin.qq.com/wxa/getwxacodeunlimit?access_token=" + access_token;
            String scene = machineNo;  //携带参数放在scene 内
            Map<String, String> param = new HashMap<>();
            param.put("scene", scene);
            //这里的page如果没有的话可以不写，默认是跳主页，如果写了没有的页面的话，会返回错误信息
//            param.put("page", "pages/index/index");
            String json = JSON.toJSONString(param);
            ByteArrayInputStream inputStream = sendPost(url, json);
            //这里判断的是返回的图片还是错误信息，一般错误信息不会大于200
            if (inputStream.available() <= 200) {
                ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
                int i;
                byte[] buffer = new byte[200];
                while ((i = inputStream.read(buffer)) != -1) {
                    byteArrayOutputStream.write(buffer, 0, i);
                }
                String str = new String(byteArrayOutputStream.toByteArray());
                //错误信息的格式在官方文档里有
                JSONObject jsonObject = JSONObject.parseObject(str);
                if ("41030".equals(jsonObject.getString("errcode"))) {
                    log.error("所传page页面不存在，或者小程序没有发布");
                    throw new CustomizeException("所传page页面不存在，或者小程序没有发布");
                } else if ("45009".equals(jsonObject.getString("errcode"))) {
                    log.error("调用分钟频率受限");
                    throw new CustomizeException("调用分钟频率受限");
                }
                byteArrayOutputStream.close();
            }
            //上传阿里云，也可以不上传
            //String aliYunUrl = ossClientUtil.UploadImgAndReturnImgUrlInputStream(inputStream, fileName, path);
            aliYunUrl = OSSUtils.uploadFileToOss("filename", inputStream, "QRCode");
            //输出到本地的代码
            FileOutputStream fileOutputStream = new FileOutputStream("D:/123.png");
            int i;
            byte[] buffer = new byte[200];
            while ((i = inputStream.read(buffer)) != -1) {
                fileOutputStream.write(buffer, 0, i);
            }
            fileOutputStream.flush();
            fileOutputStream.close();

            inputStream.close();
            log.info("二维码生成成功:{}", aliYunUrl);
        } catch (Exception e) {
            log.error("获取二维码异常");
            throw new CustomizeException(e.getMessage());
        }
        return aliYunUrl;
    }

    public static void main(String[] args) throws CustomizeException {
        String wxCode = getWXCode("name=111");
        System.out.println(wxCode);
    }
}
